package com.wen.statistics.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.wen.commonutils.ResultEntity;
import com.wen.statistics.api.UcenterApi;
import com.wen.statistics.entity.Daily;
import com.wen.statistics.mapper.DailyMapper;
import com.wen.statistics.service.DailyService;
import org.apache.commons.lang3.RandomUtils;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * <p>
 * 网站统计日数据 服务实现类
 * </p>
 *
 * @author wen
 * @since 2021-06-22
 */
@Service
public class DailyServiceImpl extends ServiceImpl<DailyMapper, Daily> implements DailyService {
    @Resource
    private UcenterApi ucenterApi;

    /**
     * 统计某一天的注册人数
     * 根据传入的day查询数据库中是否存在相同记录，如果存在，则执行更新操作，否则，执行新增操作
     * 还可以先删除数据库中日期相同的数据，然后再添加最新的记录
     *
     * @param day target day
     */
    @Override
    public void countRegister(String day) {
//        远程调用service-ucenter模块的方法，获取当前的注册人数
        ResultEntity resultEntity = ucenterApi.countRegister(day);
//        某一天的注册人数
        int countRegister = (int) resultEntity.getData().get("countRegister");
//        根据day查询数据库中是否存在相同记录
        QueryWrapper<Daily> wrapper = new QueryWrapper<>();
        wrapper.eq("date_calculated", day);
        Daily selectOne = baseMapper.selectOne(wrapper);
        if (selectOne != null) {
//            执行更新方法
            selectOne.setRegisterNum(countRegister);
            baseMapper.updateById(selectOne);
        } else {
//            执行新增方法
            Daily daily = new Daily();
            daily.setDateCalculated(day);//统计日期
            daily.setRegisterNum(countRegister);//注册人数
            daily.setCourseNum(RandomUtils.nextInt(100, 200));//每日新增课程数
            daily.setLoginNum(RandomUtils.nextInt(100, 200));//登录数
            daily.setVideoViewNum(RandomUtils.nextInt(100, 200));//每日播放视频数
            baseMapper.insert(daily);
        }
    }

    /**
     * 图表显示
     *
     * @param type  显示因子
     * @param begin 开始时间
     * @param end   结束时间
     * @return 结果 两部分数据，日期数组和数量数组
     */
    @Override
    public Map<String, Object> showData(String type, String begin, String end) {
//        要返回的map集合
        Map<String, Object> map = new HashMap<>();
//         查询条件，在begin和end之间查询满足条件的type
        QueryWrapper<Daily> wrapper = new QueryWrapper<>();
        wrapper.between("date_calculated", begin, end);
        wrapper.select("date_calculated", type);
//        查询结果
        List<Daily> dailies = baseMapper.selectList(wrapper);
//        封装数量的集合
        List<Integer> dataList = new ArrayList<Integer>();
//        封装日期的集合
        List<String> dateList = new ArrayList<String>();
//        添加到map
        map.put("dataList", dataList);
        map.put("dateList", dateList);
//        遍历dailies分别给数量和日期的集合中设置数据
        for (Daily daily : dailies) {
            dateList.add(daily.getDateCalculated());
            switch (type) {
                case "register_num":
                    dataList.add(daily.getRegisterNum());
                    break;
                case "login_num":
                    dataList.add(daily.getLoginNum());
                    break;
                case "video_view_num":
                    dataList.add(daily.getVideoViewNum());
                    break;
                case "course_num":
                    dataList.add(daily.getCourseNum());
                    break;
                default:
                    break;
            }
        }
        return map;
    }
}
